package com.jingdianjichi.oss.util;

import com.jingdianjichi.oss.entity.FileInfo;
import io.minio.*;
import io.minio.http.Method;
import io.minio.messages.Bucket;
import io.minio.messages.Item;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.io.InputStream;
import java.util.LinkedList;
import java.util.List;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

/**
 * ClassName:MinioUtil
 * Package:com.jingdianjichi.oss.util
 * Description:
 *
 * @Author fst
 * @Create 2024/10/3 20:17
 * @Version 1.0
 */
@Component
public class MinioUtil
{
    @Resource
    private MinioClient minioClient;

    /**
     * 创建bucket桶
     * @param bucket
     * @throws Exception
     */
    public  void createBucket(String bucket) throws Exception
    {
        boolean exists = minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucket).build());

        if(!exists)
        {
            minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucket).build());
        }
    }

    /**
     * 上传文件
     * @param inputStream
     * @param bucket
     * @param objectName
     * @throws Exception
     */

    public void uploadFile(InputStream inputStream,String bucket,String objectName) throws Exception
    {
        minioClient.putObject(PutObjectArgs.builder().bucket(bucket).object(objectName).stream(inputStream,-1,5242889L).build());
    }

    /**
     * 列出所有桶
     * @return
     * @throws Exception
     */
    public List<String> getAllBucket()throws Exception
    {
        List<Bucket> bucketList=minioClient.listBuckets();
        return bucketList.stream().map(Bucket::name).collect(Collectors.toList());
    }

    /**
     * 查询桶中的文件
      * @param bucket
     * @return
     * @throws Exception
     */
    public List<FileInfo> getAllFile(String bucket)throws Exception
    {
        Iterable<Result<Item>> results = minioClient.listObjects(ListObjectsArgs.builder().bucket(bucket).build());

        List<FileInfo> fileInfoList =new LinkedList<>();

        for (Result<Item> result : results) {
            Item item = result.get();
            FileInfo fileInfo=new FileInfo();

            fileInfo.setFileName(item.objectName());
            fileInfo.setDirectoryFlag(item.isDir());
            fileInfo.setEtag(item.etag());

            fileInfoList.add(fileInfo);
        }

        return fileInfoList;
    }

    /**
     * 下载文件
     * @param bucket
     * @param objectName
     * @return
     * @throws Exception
     */

    public InputStream downLoad(String bucket,String objectName) throws Exception
    {
        return minioClient.getObject(GetObjectArgs.builder().bucket(bucket).object(objectName).build());
    }


    /**
     * 删除文件
     * @param bucket
     * @param objectName
     */
    public void deleteObject(String bucket,String objectName)throws Exception
    {
        minioClient.removeObject(RemoveObjectArgs.builder().bucket(bucket).object(objectName).build());
    }

    /**
     * 删除桶
     * @param bucket
     * @throws Exception
     */

    public void deleteBucket(String bucket) throws Exception
    {
        minioClient.removeBucket(RemoveBucketArgs.builder().bucket(bucket).build());
    }

    public String getPreviewFileUrl(String bucketName, String objectName) throws Exception{
        GetPresignedObjectUrlArgs args = GetPresignedObjectUrlArgs.builder()
                .method(Method.GET)
                .bucket(bucketName).object(objectName).build();
        return minioClient.getPresignedObjectUrl(args);
    }

}
